package com.example.coursehub.classmanagement.repository;

import com.example.coursehub.classmanagement.entity.StudentClass;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.util.List;
import java.util.Optional;

/**
 * 学生班级关联数据访问接口
 */
@Repository
public interface StudentClassRepository extends JpaRepository<StudentClass, Long> {
    
    /**
     * 根据班级ID查找学生
     */
    List<StudentClass> findByClassId(Long classId);
    
    /**
     * 根据学生ID查找班级
     */
    List<StudentClass> findByStudentId(Long studentId);
    
    /**
     * 根据学生ID和班级ID查找关联记录
     */
    Optional<StudentClass> findByStudentIdAndClassId(Long studentId, Long classId);
    
    /**
     * 根据班级ID和状态查找学生
     */
    List<StudentClass> findByClassIdAndStatus(Long classId, StudentClass.StudentStatus status);
    
    /**
     * 根据学生ID和状态查找班级
     */
    List<StudentClass> findByStudentIdAndStatus(Long studentId, StudentClass.StudentStatus status);
    
    /**
     * 统计班级的学生数量（按状态）
     */
    @Query("SELECT COUNT(sc) FROM StudentClass sc WHERE sc.classId = :classId AND sc.status = :status")
    long countByClassIdAndStatus(@Param("classId") Long classId, @Param("status") StudentClass.StudentStatus status);
    
    /**
     * 统计学生参加的班级数量（按状态）
     */
    @Query("SELECT COUNT(sc) FROM StudentClass sc WHERE sc.studentId = :studentId AND sc.status = :status")
    long countByStudentIdAndStatus(@Param("studentId") Long studentId, @Param("status") StudentClass.StudentStatus status);
    
    /**
     * 检查学生是否已加入班级
     */
    boolean existsByStudentIdAndClassIdAndStatus(Long studentId, Long classId, StudentClass.StudentStatus status);
    
    /**
     * 根据班级ID删除所有学生关联
     */
    void deleteByClassId(Long classId);
}